package com.mx.zb.mng.utils;

import java.security.SecureRandom;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;

import org.apache.tomcat.util.codec.binary.Base64;

/**
 * 加密解密类，提供DES和AES两种方式
 * 
 * 推荐使用AES的方式，速度和安全性都更好
 * 
 * 关于随机数算法SHA1PRNG
 * 注意算法最后3个字母是RNG，虽然JAVA中很多内置的算法，但是能用于随机数的就是WINDOWS和LINUX都各2种
 * 而这各两种中，只有SHA1PRNG是都有的，也就是说，如果需要跨平台支持相同的加密算法，那么需要
 * 用SHA1PRNG来获得加密随机数
 * 
 * @author zb
 *
 */
public class Crypt {
	
	private static String default_pass="5kVslwcak1Jbe-SU70OpHHD";
	
	/**
	 * 将byte数组转换为表示16进制值的字符串， 如：byte[]{8,18}转换为：0813， 和public static byte[]
	 * hexStr2ByteArr(String strIn) 互为可逆的转换过程
	 * 
	 * @param arrB
	 *            需要转换的byte数组
	 * @return 转换后的字符串
	 * @throws Exception
	 *             本方法不处理任何异常，所有异常全部抛出
	 */
	public static String byteArr2HexStr(byte[] arrB) {
		int iLen = arrB.length;
		// 每个byte用两个字符才能表示，所以字符串的长度是数组长度的两倍
		StringBuffer sb = new StringBuffer(iLen * 2);
		for (int i = 0; i < iLen; i++) {
			int intTmp = arrB[i];
			// 把负数转换为正数
			while (intTmp < 0) {
				intTmp = intTmp + 256;
			}
			// 小于0F的数需要在前面补0
			if (intTmp < 16) {
				sb.append("0");
			}
			sb.append(Integer.toString(intTmp, 16));
		}
		return sb.toString();
	}

	/**
	 * 将表示16进制值的字符串转换为byte数组， 和public static String byteArr2HexStr(byte[] arrB)
	 * 互为可逆的转换过程
	 * 
	 * @param strIn
	 *            需要转换的字符串
	 * @return 转换后的byte数组
	 * @throws Exception
	 *             本方法不处理任何异常，所有异常全部抛出
	 * @author
	 */
	public static byte[] hexStr2ByteArr(String strIn) {
		byte[] arrB = strIn.getBytes();
		int iLen = arrB.length;

		// 两个字符表示一个字节，所以字节数组长度是字符串长度除以2
		byte[] arrOut = new byte[iLen / 2];
		for (int i = 0; i < iLen; i = i + 2) {
			String strTmp = new String(arrB, i, 2);
			arrOut[i / 2] = (byte) Integer.parseInt(strTmp, 16);
		}
		return arrOut;
	}
	
	
    /** 
     * 加密 
     *  
     * @param content 需要加密的内容 
     * @param password  加密密码 
     * @return 
     */  
    public static byte[] encrypt(String content, String password, String cryptType) {  
            try {             
                    KeyGenerator kgen = KeyGenerator.getInstance("AES");  
                    SecureRandom sr=SecureRandom.getInstance("SHA1PRNG");
                    sr.setSeed(password.getBytes());
                    kgen.init(128, sr);  
                    SecretKey secretKey = kgen.generateKey();  
                    byte[] enCodeFormat = secretKey.getEncoded();  
                    SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");  
                    Cipher cipher = Cipher.getInstance("AES");// 创建密码器   
                    byte[] byteContent = content.getBytes("utf-8");  
                    cipher.init(Cipher.ENCRYPT_MODE, key);// 初始化   
                    byte[] result = cipher.doFinal(byteContent);  
                    return result; // 加密   
            } catch (Exception e) {  
                  throw new RuntimeException(e.fillInStackTrace());
            }              
    }  
    
    
    /**解密 
     * @param content  待解密内容 
     * @param password 解密密钥 
     * @return 
     */  
    public static byte[] decrypt(byte[] content, String password, String cryptType) {  
            try {  
                     KeyGenerator kgen = KeyGenerator.getInstance("AES");  
                     SecureRandom sr=SecureRandom.getInstance("SHA1PRNG");
                     sr.setSeed(password.getBytes());
                     kgen.init(128, sr); 
                     SecretKey secretKey = kgen.generateKey();  
                     byte[] enCodeFormat = secretKey.getEncoded();  
                     SecretKeySpec key = new SecretKeySpec(enCodeFormat, "AES");              
                     Cipher cipher = Cipher.getInstance("AES");// 创建密码器   
                    cipher.init(Cipher.DECRYPT_MODE, key);// 初始化   
                    byte[] result = cipher.doFinal(content);  
                    return result; // 加密   
            }  catch (Exception e) {  
                throw new RuntimeException(e.fillInStackTrace());
            }   
    }
    
    /**
     * AES加密， 和PHP联调用这个加密方式
     * @param input
     * @param key
     * @return
     */
    public static String encrypt(String input, String key){  
        byte[] crypted = null;  
        try{  
	        SecretKeySpec skey = new SecretKeySpec(key.getBytes(), "AES");  
	        Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");  
	        cipher.init(Cipher.ENCRYPT_MODE, skey);  
	        crypted = cipher.doFinal(input.getBytes());  
        }catch(Exception e){  
        	throw new RuntimeException(e.fillInStackTrace());
        }  
        return new String(Base64.encodeBase64(crypted));  
    }  
    
    /**
     * AES解密， 和PHP联调用这个
     * @param input
     * @param key
     * @return
     */
    public static String decrypt(String input, String key){  
        byte[] output = null;  
        try{  
	        SecretKeySpec skey = new SecretKeySpec(key.getBytes(), "AES");  
	        Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");  
	        cipher.init(Cipher.DECRYPT_MODE, skey);  
	        output = cipher.doFinal(Base64.decodeBase64(input));  
        }catch(Exception e){  
        	throw new RuntimeException(e.fillInStackTrace());
        }  
        return new String(output);  
    }  
    
    /**
     * 用AES加密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String enAes(String content,String password) {
    	return byteArr2HexStr(encrypt(content,password, "AES"));
    }
    
    /**
     * 用AES解密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String deAes(String content){
    	return new String(decrypt(hexStr2ByteArr(content),default_pass, "AES"));
    }
    
    /**
     * 用AES加密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String enAes(String content) {
    	return byteArr2HexStr(encrypt(content,default_pass, "AES"));
    }
    
    /**
     * 用AES解密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String deAes(String content,String password){
    	return new String(decrypt(hexStr2ByteArr(content),password, "AES"));
    }
    
    /**
     * 用DES加密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String enDes(String content,String password) {
    	return byteArr2HexStr(encrypt(content,password, "DES"));
    }
    
    /**
     * 用DES解密
     * @param content
     * @param password
     * @return
     * @throws Exception 
     */
    public static String deDes(String content,String password) {
    	return new String(decrypt(hexStr2ByteArr(content),password, "DES"));
    }
    
        
    /**
     * 20150325 1000次运行测试结果
     * ==============des 1000 total use ms:506 average use ms:0
	   ==============aes 1000 total use ms:174 average use ms:0
     * @param args
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {
    	
//    	System.out.println(deDes("4ebbfdd206b8fb61b36a1675df78097c","staffByZb"));
    	
    	System.out.println(enAes("123456"));
    	
    	System.out.println(deAes("df999fcdd75d29127a0683b7c8aa27af9f28ed302a706c8a5d3120cbd76be0e7"));
    	
//    	UnionCode.getShortUuid();
//    	int runTimes=1000;
//    	
//    	long start=System.currentTimeMillis();
//		
//    	for(int i=0;i<runTimes;i++){
//    		testDes();
//    	}
//    	
//    	long end=System.currentTimeMillis();
//    	System.err.println(runTimes+"==============des total use ms:"+(end-start)+" average use ms:"+(end-start)/runTimes);
//    	
//    	start=System.currentTimeMillis();
//    	for(int i=0;i<runTimes;i++){
//    		testAes();
//    	}
//    	
//    	end=System.currentTimeMillis();
//    	System.err.println(runTimes+"==============aes total use ms:"+(end-start)+" average use ms:"+(end-start)/runTimes);
    	
	}
    
}
